Desatando el Poder de tu Django Admin: Acciones Personalizadas Explicadas

            # myapp/models.py
from django.db import models

class Product(models.Model):
    name = models.CharField(max_length=200)
    price = models.DecimalField(max_digits=10, decimal_places=2)
    is_discounted = models.BooleanField(default=False)
    created_at = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return self.name

            

              
                Copy
              
            
          

            # myapp/admin.py
from django.contrib import admin, messages
from django.db.models import QuerySet
from django.http import HttpRequest
from .models import Product

@admin.register(Product)
class ProductAdmin(admin.ModelAdmin):
    list_display = ('name', 'price', 'is_discounted', 'created_at')
    list_filter = ('is_discounted', 'created_at')
    search_fields = ('name',)

    # Define la función de acción personalizada del admin
    def make_discounted(self, request: HttpRequest, queryset: QuerySet):
        updated_count = queryset.update(is_discounted=True)
        self.message_user(
            request,
            f"{updated_count} producto(s) fueron marcados exitosamente como descontados.",
            messages.SUCCESS
        )
    make_discounted.short_description = "Marcar productos seleccionados como descontados"

    # Registra la acción con el ModelAdmin
    actions = [make_discounted]

            

              
                Copy
              
            
          

            # myapp/models.py (asumiendo un modelo Post)
from django.db import models

class Post(models.Model):
    title = models.CharField(max_length=255)
    content = models.TextField()
    status = models.CharField(max_length=20, default='draft', choices=[
        ('draft', 'Draft'),
        ('pending', 'Pending Review'),
        ('approved', 'Approved'),
        ('rejected', 'Rejected'),
    ])
    created_at = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return self.title

            

              
                Copy
              
            
          

            # myapp/forms.py
from django import forms

class ConfirmationForm(forms.Form):
    confirm = forms.BooleanField(
        label="¿Estás seguro de que quieres realizar esta acción?",
        required=True,
        widget=forms.HiddenInput # Manejaremos la visualización en la plantilla
    )
    _selected_action = forms.CharField(widget=forms.HiddenInput)
    action = forms.CharField(widget=forms.HiddenInput)

            

              
                Copy
              
            
          

            # myapp/admin.py
from django.contrib import admin, messages
from django.db.models import QuerySet
from django.http import HttpRequest, HttpResponseRedirect
from django.shortcuts import render
from django.urls import reverse
from django.utils.translation import gettext_lazy as _
from .models import Post
from .forms import ConfirmationForm

@admin.register(Post)
class PostAdmin(admin.ModelAdmin):
    list_display = ('title', 'status', 'created_at')
    list_filter = ('status',)
    search_fields = ('title',)

    def mark_posts_approved(self, request: HttpRequest, queryset: QuerySet) -> HttpResponseRedirect | None:
        # Comprueba si el usuario confirmó la acción
        if 'apply' in request.POST:
            form = ConfirmationForm(request.POST)
            if form.is_valid():
                updated_count = queryset.update(status='approved')
                self.message_user(
                    request,
                    f"{updated_count} post(s) fueron marcados exitosamente como aprobados.",
                    messages.SUCCESS
                )
                return HttpResponseRedirect(request.get_full_path())
        
        # Si no se confirma, o es una solicitud GET, muestra la página de confirmación
        else:
            # Almacena las claves primarias de los objetos seleccionados en un campo oculto
            # Esto es crucial para pasar la selección a través de la página de confirmación
            context = self.admin_site.each_context(request)
            context['queryset'] = queryset
            context['form'] = ConfirmationForm(initial={
                '_selected_action': request.POST.getlist(admin.ACTION_CHECKBOX_NAME),
                'action': 'mark_posts_approved',
            })
            context['action_name'] = self.mark_posts_approved.short_description
            context['title'] = _("Confirmar Acción")

            # Renderiza una plantilla de confirmación personalizada
            return render(request, 'admin/confirmation_action.html', context)

    mark_posts_approved.short_description = _("Marcar posts seleccionados como aprobados")

    actions = [mark_posts_approved]

            

              
                Copy
              
            
          

            {# templates/admin/confirmation_action.html #}
{% extends "admin/base_site.html" %}
{% load i18n admin_urls static admin_modify %}

{% block extrastyle %}{{ block.super }}
    
{% endblock %}

{% block content %}
    

        

            {% csrf_token %}
            
{{ action_name }}
            
Estás a punto de realizar la acción "{{ action_name }}" sobre los siguientes {{ queryset.count }} objeto(s):
            

                {% for obj in queryset %}
                    
{{ obj }}
                {% endfor %}
            
            
¿Estás absolutamente seguro de que deseas continuar?

            {{ form.as_p }}

            
            {% for item in form._selected_action.value %}
                
            {% endfor %}

            

                
                
            
        
    
{% endblock %}

            

              
                Copy
              
            
          

            # myapp/forms.py
from django import forms
from .models import Post

class ChangePostStatusForm(forms.Form):
    _selected_action = forms.CharField(widget=forms.HiddenInput)
    action = forms.CharField(widget=forms.HiddenInput)

    new_status = forms.ChoiceField(
        label="Nuevo Estado",
        choices=Post.STATUS_CHOICES, # Asumiendo STATUS_CHOICES definido en el modelo Post
        required=True
    )
    comment = forms.CharField(
        label="Razón/Comentario (opcional)",
        required=False,
        widget=forms.Textarea(attrs={'rows': 3})
    )

# Agrega STATUS_CHOICES al modelo Post
# myapp/models.py
from django.db import models

class Post(models.Model):
    STATUS_CHOICES = [
        ('draft', 'Draft'),
        ('pending', 'Pending Review'),
        ('approved', 'Approved'),
        ('rejected', 'Rejected'),
    ]
    title = models.CharField(max_length=255)
    content = models.TextField()
    status = models.CharField(max_length=20, default='draft', choices=STATUS_CHOICES)
    comment_history = models.TextField(blank=True, null=True)
    created_at = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return self.title

            

              
                Copy
              
            
          

            # myapp/admin.py (continuación)
from django.contrib import admin, messages
from django.db.models import QuerySet
from django.http import HttpRequest, HttpResponseRedirect
from django.shortcuts import render
from django.urls import reverse
from django.utils.translation import gettext_lazy as _
from .models import Post
from .forms import ChangePostStatusForm # Importa el nuevo formulario

@admin.register(Post)
class PostAdmin(admin.ModelAdmin):
    list_display = ('title', 'status', 'created_at')
    list_filter = ('status',)
    search_fields = ('title',)

    # Acción existente mark_posts_approved...

    def change_post_status_with_comment(self, request: HttpRequest, queryset: QuerySet) -> HttpResponseRedirect | None:
        form = None
        if 'apply' in request.POST:
            form = ChangePostStatusForm(request.POST)
            if form.is_valid():
                new_status = form.cleaned_data['new_status']
                comment = form.cleaned_data['comment']

                updated_count = 0
                for post in queryset:
                    post.status = new_status
                    if comment:
                        post.comment_history = (post.comment_history or '') + f"\n[{request.user.username}] changed to {new_status} with comment: {comment}"
                    post.save()
                    updated_count += 1

                self.message_user(
                    request,
                    f"{updated_count} post(s) tuvieron su estado cambiado a '{new_status}' y se agregó un comentario.",
                    messages.SUCCESS
                )
                return HttpResponseRedirect(request.get_full_path())
        
        # Si no se confirma, o es una solicitud GET, muestra el formulario de entrada
        if not form:
            form = ChangePostStatusForm(initial={
                '_selected_action': request.POST.getlist(admin.ACTION_CHECKBOX_NAME),
                'action': 'change_post_status_with_comment',
            })

        context = self.admin_site.each_context(request)
        context['queryset'] = queryset
        context['form'] = form
        context['action_name'] = self.change_post_status_with_comment.short_description
        context['title'] = _("Cambiar Estado del Post y Agregar Comentario")

        return render(request, 'admin/change_status_action.html', context)

    change_post_status_with_comment.short_description = _("Cambiar estado para posts seleccionados (con comentario)")

    actions = [
        mark_posts_approved,
        change_post_status_with_comment
    ]

            

              
                Copy
              
            
          

            {# templates/admin/change_status_action.html #}
{% extends "admin/base_site.html" %}
{% load i18n admin_urls static admin_modify %}

{% block extrastyle %}{{ block.super }}
    
{% endblock %}

{% block content %}
    

        

            {% csrf_token %}
            
{{ action_name }}
            
Estás a punto de cambiar el estado para los siguientes {{ queryset.count }} objeto(s):
            

                {% for obj in queryset %}
                    
{{ obj }}
                {% endfor %}
            
            
Por favor, selecciona el nuevo estado y opcionalmente agrega un comentario:

            {% for field in form %}
                {% if field.is_hidden %}
                    {{ field }}
                {% else %}
                    

                        {{ field.errors }}
                        {{ field.label_tag }}
                        {{ field }}
                        {% if field.help_text %}
{{ field.help_text }}
{% endif %}
                    
                {% endif %}
            {% endfor %}

            

                
                
            
        
    
{% endblock %}

            

              
                Copy
              
            
          

            # myapp/admin.py (continuación)
import requests # Necesitarás instalar requests: pip install requests
# ... otras importaciones ...

# Asumiendo ProductAdmin de antes

class ProductAdmin(admin.ModelAdmin):
    # ... list_display existente, list_filter, search_fields ...

    def sync_products_to_external_ims(self, request: HttpRequest, queryset: QuerySet) -> HttpResponseRedirect | None:
        # Comprueba la confirmación (similar a ejemplos anteriores si es necesario)
        if 'apply' in request.POST:
            # Simula un endpoint de API externo
            EXTERNAL_IMS_API_URL = "https://api.example.com/v1/products/sync/"
            API_KEY = "tu_clave_api_secreta" # En una app real, usa settings.py o variables de entorno

            successful_syncs = 0
            failed_syncs = []

            for product in queryset:
                data = {
                    "product_id": product.id,
                    "name": product.name,
                    "price": str(product.price), # Convierte Decimal a string para JSON
                    "is_discounted": product.is_discounted,
                    # Agrega otros datos relevantes del producto
                }
                headers = {
                    "Authorization": f"Bearer {API_KEY}",
                    "Content-Type": "application/json"
                }
                
                try:
                    response = requests.post(EXTERNAL_IMS_API_URL, json=data, headers=headers, timeout=5) # Timeout de 5 segundos
                    response.raise_for_status() # Lanza un HTTPError para respuestas incorrectas (4xx o 5xx)
                    successful_syncs += 1
                except requests.exceptions.RequestException as e:
                    failed_syncs.append(f"Producto {product.name} (ID: {product.id}): {e}")
                except Exception as e:
                    failed_syncs.append(f"Producto {product.name} (ID: {product.id}): error inesperado: {e}")

            if successful_syncs > 0:
                self.message_user(
                    request,
                    f"{successful_syncs} producto(s) sincronizados exitosamente con el IMS externo.",
                    messages.SUCCESS
                )
            if failed_syncs:
                error_message = f"No se pudieron sincronizar {len(failed_syncs)} producto(s):\n" + "\n".join(failed_syncs)
                self.message_user(request, error_message, messages.ERROR)
            
            return HttpResponseRedirect(request.get_full_path())

        # Solicitud GET inicial o solicitud POST no-apply: muestra confirmación (si se desea)
        context = self.admin_site.each_context(request)
        context['queryset'] = queryset
        context['form'] = ConfirmationForm(initial={
            '_selected_action': request.POST.getlist(admin.ACTION_CHECKBOX_NAME),
            'action': 'sync_products_to_external_ims',
        })
        context['action_name'] = self.sync_products_to_external_ims.short_description
        context['title'] = _("Confirmar Sincronización de Datos")

        return render(request, 'admin/confirmation_action.html', context) # Reutiliza la plantilla de confirmación

    sync_products_to_external_ims.short_description = _("Sincronizar productos seleccionados con IMS externo")

    actions = [
        # ... otras acciones ...
        sync_products_to_external_ims,
    ]

            

              
                Copy
              
            
          

            # myapp/admin.py (continuación)
import csv
from django.http import HttpResponse
# ... otras importaciones ...

class PostAdmin(admin.ModelAdmin):
    # ... list_display existente, list_filter, search_fields, actions ...

    def export_posts_as_csv(self, request: HttpRequest, queryset: QuerySet) -> HttpResponse:
        response = HttpResponse(content_type='text/csv')
        response['Content-Disposition'] = 'attachment; filename="posts_export.csv"'

        writer = csv.writer(response)
        # Escribe la fila de encabezado
        writer.writerow(['Título', 'Estado', 'Fecha de Creación', 'Vista Previa del Contenido'])

        for post in queryset:
            writer.writerow([
                post.title,
                post.get_status_display(), # Usa get_FOO_display() para campos de opción
                post.created_at.strftime("%Y-%m-%d %H:%M:%S"),
                post.content[:100] + '...' if len(post.content) > 100 else post.content # Trunca contenido largo
            ])
        
        self.message_user(
            request,
            f"{queryset.count()} post(s) exportados exitosamente a CSV.",
            messages.SUCCESS
        )
        return response

    export_posts_as_csv.short_description = _("Exportar posts seleccionados como CSV")

    actions = [
        # ... otras acciones ...
        export_posts_as_csv,
    ]

            

              
                Copy
              
            
          

            # myproject/celery.py (configuración estándar de Celery)
import os
from celery import Celery

# Establece el módulo de configuración predeterminado de Django para el programa 'celery'.
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'myproject.settings')

app = Celery('myproject')

# Usar una cadena aquí significa que el trabajador no tendrá que
# serializar el objeto al usar Windows.
app.config_from_object('django.conf:settings', namespace='CELERY')

# Carga los módulos de tareas de todas las configuraciones de aplicaciones registradas de Django.
app.autodiscover_tasks()

@app.task(bind=True)
def debug_task(self):
    print(f'Solicitud: {self.request!r}')

            

              
                Copy
              
            
          

            # myapp/tasks.py
import requests
from celery import shared_task
from django.contrib.auth import get_user_model
from django.apps import apps

@shared_task
def sync_product_to_external_ims_task(product_id, admin_user_id):
    Product = apps.get_model('myapp', 'Product')
    User = get_user_model()

    try:
        product = Product.objects.get(pk=product_id)
        admin_user = User.objects.get(pk=admin_user_id)

        EXTERNAL_IMS_API_URL = "https://api.example.com/v1/products/sync/"
        API_KEY = "tu_clave_api_secreta" # Usa variables de entorno o configuración de Django

        data = {
            "product_id": product.id,
            "name": product.name,
            "price": str(product.price),
            "is_discounted": product.is_discounted,
        }
        headers = {
            "Authorization": f"Bearer {API_KEY}",
            "Content-Type": "application/json"
        }

        response = requests.post(EXTERNAL_IMS_API_URL, json=data, headers=headers, timeout=10)
        response.raise_for_status()
        
        # Registra el éxito (por ejemplo, en logs de Django o un modelo específico para seguimiento)
        print(f"Producto {product.name} (ID: {product.id}) sincronizado por {admin_user.username} exitosamente.")

    except Product.DoesNotExist:
        print(f"Producto con ID {product_id} no encontrado.")
    except User.DoesNotExist:
        print(f"Usuario admin con ID {admin_user_id} no encontrado.")
    except requests.exceptions.RequestException as e:
        print(f"Fallo de sincronización de API para el producto {product_id}: {e}")
    except Exception as e:
        print(f"Error inesperado durante la sincronización para el producto {product_id}: {e}")


            

              
                Copy
              
            
          

            # myapp/admin.py (continuación)
from django.contrib import admin, messages
from django.db.models import QuerySet
from django.http import HttpRequest, HttpResponseRedirect
from django.shortcuts import render
from django.urls import reverse
from django.utils.translation import gettext_lazy as _
from .models import Product # Asumiendo el modelo Product de antes
from .tasks import sync_product_to_external_ims_task # Importa tu tarea Celery

class ProductAdmin(admin.ModelAdmin):
    # ... list_display existente, list_filter, search_fields ...

    def async_sync_products_to_external_ims(self, request: HttpRequest, queryset: QuerySet) -> HttpResponseRedirect | None:
        if 'apply' in request.POST:
            admin_user_id = request.user.id
            for product in queryset:
                # Encola la tarea para cada producto seleccionado
                sync_product_to_external_ims_task.delay(product.id, admin_user_id)
            
            self.message_user(
                request,
                f"Las tareas de sincronización de {queryset.count()} producto(s) han sido puestas en cola.",
                messages.SUCCESS
            )
            return HttpResponseRedirect(request.get_full_path())

        # Solicitud GET inicial o solicitud POST no-apply: muestra confirmación
        context = self.admin_site.each_context(request)
        context['queryset'] = queryset
        context['form'] = ConfirmationForm(initial={
            '_selected_action': request.POST.getlist(admin.ACTION_CHECKBOX_NAME),
            'action': 'async_sync_products_to_external_ims',
        })
        context['action_name'] = self.async_sync_products_to_external_ims.short_description
        context['title'] = _("Confirmar Sincronización de Datos Asíncrona")

        return render(request, 'admin/confirmation_action.html', context) # Reutiliza la plantilla de confirmación

    async_sync_products_to_external_ims.short_description = _("Poner en cola sincronización asíncrona para productos seleccionados al IMS")

    actions = [
        # ... otras acciones ...
        async_sync_products_to_external_ims,
    ]

            

              
                Copy
              
            
          

            def sensitive_action(self, request, queryset):
    if not request.user.has_perm('myapp.can_perform_sensitive_action'):
        self.message_user(request, _("No tienes permiso para realizar esta acción."), messages.ERROR)
        return HttpResponseRedirect(request.get_full_path())
    # ... lógica de acción sensible ...

            

              
                Copy
              
            
          

            # myapp/models.py
class Product(models.Model):
    # ... campos ...
    class Meta:
        permissions = [
            ("can_perform_sensitive_action", "Can perform sensitive product action"),
        ]

            

              
                Copy
              
            
          

            # myapp/admin.py
class ProductAdmin(admin.ModelAdmin):
    # ... definición de acciones ...

    def get_actions(self, request: HttpRequest):
        actions = super().get_actions(request)
        # Elimina la acción 'make_discounted' si el usuario no tiene un permiso específico
        if not request.user.has_perm('myapp.change_product'): # O un permiso personalizado como 'can_discount_product'
            if 'make_discounted' in actions:
                del actions['make_discounted']
        return actions

            

              
                Copy
              
            
          

            # myapp/admin_actions.py
from django.contrib import messages
from django.http import HttpRequest, HttpResponseRedirect
from django.db.models import QuerySet

def mark_as_active(modeladmin, request: HttpRequest, queryset: QuerySet):
    updated = queryset.update(is_active=True)
    modeladmin.message_user(request, f"{updated} elemento(s) marcados como activos.", messages.SUCCESS)
mark_as_active.short_description = "Marcar seleccionados como activos"

            

              
                Copy
              
            
          

            # myapp/admin.py
from django.contrib import admin
from .models import MyModel
from .admin_actions import mark_as_active

@admin.register(MyModel)
class MyModelAdmin(admin.ModelAdmin):
    list_display = ('name', 'is_active')
    actions = [mark_as_active]

            

              
                Copy
              
            
          

            # myapp/tests.py
from django.test import TestCase, Client
from django.contrib.auth import get_user_model
from django.urls import reverse
from django.contrib import admin
from .models import Product
from .admin import ProductAdmin # Importa tu ModelAdmin

User = get_user_model()

class ProductAdminActionTests(TestCase):
    def setUp(self):
        self.admin_user = User.objects.create_superuser('admin', 'admin@example.com', 'password')
        self.client = Client()
        self.client.login(username='admin', password='password')

        self.p1 = Product.objects.create(name="Product A", price=10.00, is_discounted=False)
        self.p2 = Product.objects.create(name="Product B", price=20.00, is_discounted=False)
        self.p3 = Product.objects.create(name="Product C", price=30.00, is_discounted=True)

        self.admin_site = admin.AdminSite()
        self.model_admin = ProductAdmin(Product, self.admin_site)

    def test_make_discounted_action(self):
        # Simula la selección de productos y la ejecución de la acción
        change_list_url = reverse('admin:myapp_product_changelist')
        response = self.client.post(change_list_url, {
            admin.ACTION_CHECKBOX_NAME: [self.p1.pk, self.p2.pk],
            'action': 'make_discounted',
            'index': 0, # Requerido para cierta lógica interna del admin de Django
        }, follow=True)

        self.assertEqual(response.status_code, 200)
        self.assertContains(response, '2 producto(s) fueron marcados exitosamente como descontados.')
        
        self.p1.refresh_from_db()
        self.p2.refresh_from_db()
        self.p3.refresh_from_db()

        self.assertTrue(self.p1.is_discounted)
        self.assertTrue(self.p2.is_discounted)
        self.assertTrue(self.p3.is_discounted) # Este ya estaba descontado

    def test_make_discounted_action_confirmation(self):
        # Para acciones con confirmación, probarías el proceso de dos pasos
        change_list_url = reverse('admin:myapp_post_changelist') # Asumiendo modelo Post para el ejemplo de confirmación
        post1 = Post.objects.create(title='Test Post 1', content='...', status='draft')
        post2 = Post.objects.create(title='Test Post 2', content='...', status='draft')

        # Paso 1: Solicitar página de confirmación
        response = self.client.post(change_list_url, {
            admin.ACTION_CHECKBOX_NAME: [post1.pk, post2.pk],
            'action': 'mark_posts_approved',
            'index': 0,
        })
        self.assertEqual(response.status_code, 200)
        self.assertIn(b"Confirmar Acción", response.content) # Comprueba si se renderiza la página de confirmación

        # Paso 2: Enviar formulario de confirmación
        response = self.client.post(change_list_url, {
            admin.ACTION_CHECKBOX_NAME: [post1.pk, post2.pk],
            'action': 'mark_posts_approved',
            'apply': 'Sí, estoy seguro',
            'confirm': 'on', # Valor para un checkbox si se renderiza como checkbox
            '_selected_action': [str(post1.pk), str(post2.pk)], # Debe pasarse de vuelta desde el formulario
            'index': 0,
        }, follow=True)

        self.assertEqual(response.status_code, 200)
        self.assertContains(response, '2 post(s) fueron marcados exitosamente como aprobados.')

        post1.refresh_from_db()
        post2.refresh_from_db()
        self.assertEqual(post1.status, 'approved')
        self.assertEqual(post2.status, 'approved')

            

              
                Copy